package org.cll.lagou.study.ssm.controller;

import org.apache.commons.lang3.StringUtils;
import org.cll.lagou.study.ssm.entity.User;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.stereotype.Controller;
import org.springframework.ui.Model;
import org.springframework.ui.ModelMap;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.servlet.support.RequestContextUtils;

import javax.imageio.ImageIO;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import javax.servlet.http.HttpSession;
import java.awt.*;
import java.awt.image.BufferedImage;
import java.io.IOException;
import java.util.Map;
import java.util.Random;

/**
 * 描述信息:
 *
 * @author CLL
 * @version 1.0
 * @date 2020/6/22 17:39
 */
@Controller
@RequestMapping(value = "/user")
public class UserController {

    private static final Logger logger = LoggerFactory.getLogger(UserController.class);

    @RequestMapping(value = "/login")
    public String login(ModelMap modelMap, User user, HttpSession httpSession) {
        // 获取重定向参数 方式1
//        Map<String, Object> modelMap = (Map<String, Object>) RequestContextUtils.getInputFlashMap(request);
        if (modelMap.containsAttribute("init")) {
            return "login";
        }
        if (null == user || StringUtils.isBlank(user.getUsername()) || StringUtils.isBlank(user.getPassword())) {
            modelMap.addAttribute("msg", "用户名和密码不能为空");
            return "login";
        }
        boolean flag = "admin".equals(user.getUsername()) && "admin".equals(user.getPassword());
        if (!flag) {
            modelMap.addAttribute("msg", "用户名或密码错误");
            return "login";
        }
        httpSession.setAttribute("user", user);
        return "redirect:/resume/list";
    }

    /**
     * 生成验证码
     * @param response  响应
     * @param session   会话
     * @throws IOException  IO异常
     */
    @RequestMapping("/authCode")
    public void getAuthCode(HttpServletResponse response, HttpSession session)  throws IOException {
        int width = 100;
        int height = 45;
        Random random = new Random();
        //设置response头信息
        //禁止缓存
        response.setHeader("Pragma", "No-cache");
        response.setHeader("Cache-Control", "no-cache");
        response.setDateHeader("Expires", 0);

        //生成缓冲区image类
        BufferedImage image = new BufferedImage(width, height, 1);
        //产生image类的Graphics用于绘制操作
        Graphics g = image.getGraphics();
        //Graphics类的样式
        g.setColor(this.getRandColor(200, 250));
        g.setFont(new Font("Times New Roman",0,32));
        g.fillRect(0, 0, width, height);
        //绘制字符
        String strCode = "";
        for(int i=0;i<4;i++){
            String rand = String.valueOf(random.nextInt(10));
            strCode = strCode + rand;
            g.setColor(new Color(20+random.nextInt(110),20+random.nextInt(110),20+random.nextInt(110)));
            //g.drawString(rand, 13*i+6, 28);
            g.drawString(rand, 15*i+8+random.nextInt(8), 30+random.nextInt(10));
        }
        //绘制干扰线
        for(int i=0;i<40;i++){
            g.setColor(this.getRandColor(130, 200));
            int x = random.nextInt(width);
            int y = random.nextInt(height);
            int x1 = random.nextInt(12);
            int y1 = random.nextInt(12);
            g.drawLine(x, y, x + x1, y + y1);
        }
        //将字符保存到session中用于前端的验证
        session.setAttribute("strCode", strCode);
        g.dispose();
        ImageIO.write(image, "JPEG", response.getOutputStream());
        response.getOutputStream().flush();
    }
    private Color getRandColor(int fc, int bc){
        Random random = new Random();
        if(fc>255) {
            fc = 255;
        }
        if(bc>255) {
            bc = 255;
        }
        int r = fc + random.nextInt(bc - fc);
        int g = fc + random.nextInt(bc - fc);
        int b = fc + random.nextInt(bc - fc);
        return new Color(r,g,b);
    }

}